#pragma once
#include<iostream>
#include<unistd.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>


#include"../comm/util.hpp"
#include"../comm/log.hpp"

// 只负责进行代码的编译
//  只负责进行代码的编译
namespace ns_compiler
{
    using namespace ns_util; // 引入路径拼接功能
    using namespace ns_log;  // 引入日志打印功能

    // 编译功能
    class Compiler
    {
    public:
        Compiler()
        {
        }

        ~Compiler()
        {
        }

        // 编译代码：
        // 返回值：编译成功：true，编译错误：false
        // 输入参数：编译的文件名：file_name
        // 例如：file_name: 12345（只传进来一个文件名，不带后缀，后缀我们自己添加）
        // 我们会生成三个文件
        // 12345 -> ./temp/12345.cpp 源文件
        // 12345 -> ./temp/12345.exe 可执行文件
        // 12345 -> ./temp/12345.stderr 标准错误文件
        static bool Compile(std::string &file_name)
        {
            pid_t pid = fork(); // 创建一个子进程进行编译代码

            if (pid < 0)
            {
                // 如果创建子进程失败就直接退出
                LOG(ERROR) << "内部错误，创建子进程失败" << "\n";
                return false;
            }
            else if (pid == 0)
            {
                umask(0); // 先将umask清零，让其不受平台的影响

                // 先将编译时错误文件打开
                int _compile_error = open(PathUtil::CompilerError(file_name).c_str(), O_CREAT | O_WRONLY, 0644);
                if (_compile_error < 0)
                {
                    // 如果打开文件失败
                    LOG(WARNING) << "没有成功形成compile_error文件" << "\n";
                    exit(1);
                }

                // 重定向标准错误到_compile_error
                dup2(_compile_error, 2);

                // 注意：程序替换，并不影响进程的文件描述符表
                // 子进程：调用编译器，完成对代码的编译工作
                // g++ -o target src -std=c++11
                execlp("g++", "g++", "-o", PathUtil::Exe(file_name).c_str(),
                       PathUtil::Src(file_name).c_str(), "-D", "COMPILER_ONLINE", "-std=c++11", nullptr /*不要忘记写nullptr*/); // 用g++替换当前的子进程

                // 如果程序替换失败
                LOG(ERROR) << "启动编译器g++失败，可能是参数错误" << "\n";
                exit(2);
            }
            else
            {
                // 父进程
                waitpid(pid, nullptr, 0);

                // 判断编译是否成功（判断是否生成可执行文件）
                if (FileUtil::IsFileExists(PathUtil::Exe(file_name)))
                {
                    LOG(INFO) << PathUtil::Src(file_name) << " 编译成功！" << "\n";
                    return true;
                }
            }

            // 如果父进程走到这里，就说明编译失败，没有形成可执行文件
            LOG(ERROR) << "编译失败，没有形成可执行文件！" << "\n";
            return false;
        }
    };
}
